#pragma once
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <atomic>
#include <fstream>
#include <vector>
#include <boost/algorithm/string.hpp>
#include <mysql/mysql.h>

namespace ns_util
{
    class TimeUtil
    {
    public:
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }
        // 获取毫秒时间
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };
    const std::string temp_path = "./temp/";

    class PathUtil
    {
    public:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }
        // 编译时需要的临时文件
        // 构建对应的源文件路径+后缀的完整文件名
        // test -> ./temp/test.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 构建可执行程序的完成路径和后缀名
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compiler_error");
        }

        // 运行时需要的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        // 构建该程序对应的标准错误完成的路径和后缀名
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };
    class FileUtil
    {
    public:
        static bool isFileExists(const std::string &path_name)
        {
            // stat 对应路径下的文件属性
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
            {
                // 获取成功，说明文件已经存在
                return true;
            }
            return false;
        }
        static std::string UniqueFileName()
        {
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(id);

            return ms + "_" + uniq_id;
        }
        static bool WriteFile(const std::string &target, const std::string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                return false;
            }
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            (*content).clear();

            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            // getline 不保存分隔符
            // getline 内部重载了强制类型转化,while循环可以判断
            while (std::getline(in, line))
            {
                (*content) += line;
                if (keep)
                {
                    (*content) += "\n";
                }
            }
            in.close();
            return true;
        }
    };
    class StringUtil
    {
    public:
        /*
        str: 输入型，目标要切分的字符串
        target：输出型，保存切分完毕的结果
        sep：指定的分隔符
        */
        static void SplitString(std::string &str, std::vector<std::string> *target, const std::string &sep)
        {
            // 切割
            boost::split(*target, str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
    class MySqlHandler
    {
    public:
        // 构造函数：初始化连接参数
        MySqlHandler(const std::string &h, const std::string &u,
                     const std::string &pwd, const std::string &db,
                     unsigned int pt) : host(h), user(u), password(pwd), dbName(db), port(pt), conn(nullptr) {}
        // 析构函数：关闭连接
        ~MySqlHandler()
        {
            disconnect();
        }
        // 连接数据库
        bool connect()
        {
            // 初始化连接句柄
            conn = mysql_init(nullptr);
            if (conn == nullptr)
            {
                // 说明初始化MySQL连接失败了
                return false;
            }
            // 连接数据库
            if (!mysql_real_connect(conn, host.c_str(), user.c_str(), password.c_str(), dbName.c_str(), port, nullptr, 0))
            {
                mysql_close(conn);
                conn = nullptr;
                return false;
            }
            return true;
        }
        void disconnect()
        {
            if (conn)
            {
                mysql_close(conn);
                conn = nullptr;
            }
        }
        // 获取表中的所有数据
        std::vector<std::vector<std::string>> getAllData(const std::string &sql)
        {
            if (mysql_query(conn, sql.c_str()) != 0)
            {
                return std::vector<std::vector<std::string>>();
            }
            // 获取结果集
            MYSQL_RES *result = mysql_store_result(conn);
            std::vector<std::vector<std::string>> data;
            MYSQL_ROW row;
            unsigned int numCols = mysql_num_fields(result);
            while ((row = mysql_fetch_row(result)) != nullptr)
            {
                std::vector<std::string> rowData;
                for (unsigned int i = 0; i < numCols; ++i)
                {
                    // 处理NULL值（转为空字符串）
                    rowData.push_back(row[i] ? row[i] : "");
                }
                data.push_back(rowData);
            }
            freeResult(result);
            return data;
        }
        
    private:
        MYSQL *conn;           // 数据库连接句柄
        std::string host;      // 主机地址
        std::string user;      // 用户名
        std::string password;  // 密码
        std::string dbName;    // 数据库名
        std::string tableName; // 当前操作的表名
        unsigned int port;     // 端口号（默认3306）
        // 释放结果集资源（内部辅助函数）
        void freeResult(MYSQL_RES *result)
        {
            if (result != nullptr)
            {
                mysql_free_result(result);
            }
        }
    };
}